from os import path
from flask import Blueprint,request,jsonify
from utile.comm import get_data
from db import db
bp_course = Blueprint('course',__name__)

#无限级分类重构
def xtree(data):
    #先判断data没有直接返回
    if not data:
        return data
    #遍历data变成字典
    tree={}
    list=[]
    for i in data:
        tree[i['id']] = i

    for j in data:
        #获取父id
        pid = j['pid']
        if pid == 0:
            #说明顶级分类直接加入列表
            list.append(j)
        else:
            #查看父类下有没有子类
            if 'son' not in tree[pid]:
                #如果没有初始化列表
                tree[pid]['son'] = []

            #把子类加到父类列表中
            tree[pid]['son'].append(j)
    # print(tree)
    return list
